import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.po.TableFill;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

import java.util.ArrayList;

public class CodeGenerator {
    public static void main(String[] args) {
        //创建一个自动生成器对象
        AutoGenerator autoGenerator = new AutoGenerator();
        //连接数据库配置
        DataSourceConfig dataSource = new DataSourceConfig();
        //设置驱动
        dataSource.setDriverName("com.mysql.jdbc.Driver");
        //数据库地址
        dataSource.setUrl("jdbc:mysql://127.0.0.1:3306/tanhua?useUnicode=true&characterEncoding=utf8");
        //数据库账号
        dataSource.setUsername("root");
        //数据库密码
        dataSource.setPassword("root");
        //为自动生成器对象设置数据库
        autoGenerator.setDataSource(dataSource);

        //设置全局配置
        GlobalConfig globalConfig = new GlobalConfig();
        globalConfig.setOutputDir("F:\\2020java\\tanhua-parent\\tanhua-model\\src\\main\\java");//设置代码生成位置;
        globalConfig.setOpen(false);//设置生成完毕后知否打开生成代码所在的目录
        globalConfig.setAuthor("爱吃豆的土豆");  //设置作者
        globalConfig.setFileOverride(true);//设置是否覆盖原始生成的文件
//        globalConfig.setMapperName("%s")设置数据层接口名，%s为占位符，指代模块名称
        globalConfig.setIdType(IdType.AUTO);
        globalConfig.setServiceName("%sService");
        autoGenerator.setGlobalConfig(globalConfig);


        //设置包名等相关配置
        PackageConfig packageConfig = new PackageConfig();
        packageConfig.setParent("com.czxy.tanhua");
//        packageConfig.setEntity("domian")   可以设置实体类的包名称
        //       packageConfig.setMapper("mapper")   //设置数据层包名
        autoGenerator.setPackageInfo(packageConfig);


        //策略设置
        StrategyConfig strategyConfig = new StrategyConfig();
        strategyConfig.setControllerMappingHyphenStyle(false); // RequestMapping种的驼峰是否转成用"-"连接, 默认是false
        strategyConfig.setEntityTableFieldAnnotationEnable(true); // 为实体类的类上加@TableName, 所有字段上加注解
        //strategyConfig.setInclude("med_user");  //设置当前参与生成的表名，参数为可变参数
        strategyConfig.setTablePrefix("tb_");//设置数据库表的前缀名称，模块名 = 数据库表名-前缀名 例如User = tb_user - tb_
        //设置是否开启Rest风格
        strategyConfig.setRestControllerStyle(true);
        //设置乐观锁字段名
        //strategyConfig.setVersionFieldName("");
        //设置逻辑删除字段名
        //strategyConfig.setLogicDeleteFieldName("is_deleted");
        //设置是否启用Lombok
        strategyConfig.setEntityLombokModel(true);
//        表名转换方式：数据库中的下划线转成Java驼峰
        strategyConfig.setNaming(NamingStrategy.underline_to_camel);
//        列名转换方式
        strategyConfig.setColumnNaming(NamingStrategy.underline_to_camel);
        // 自动填充配置: 插入时间, 最后一次更新时间
        ArrayList<TableFill> tableFills = new ArrayList<>();
        TableFill gmtCreate = new TableFill("created_at", FieldFill.INSERT); // 插入时改变的时间纪录, created为表的字段名
        TableFill gmtModified = new TableFill("updated_at", FieldFill.INSERT_UPDATE); // 最后一次更新时updated表字段的时间记录
        tableFills.add(gmtCreate);
        tableFills.add(gmtModified);
        strategyConfig.setTableFillList(tableFills);
        autoGenerator.setStrategy(strategyConfig);

        //执行生成

        autoGenerator.execute();
    }
}
